package cn.yunhe.util;

import org.aspectj.lang.JoinPoint;

/**
 * 拦截器方法
 */
public class DataSourceExchange {


    public void before(JoinPoint point) {
        //获取目标对象的类型
        Class<?> aClass = point.getTarget().getClass();

        //获取包名用于区分不同数据源

        String whichDataSource = aClass.getName().substring(17, aClass.getName().lastIndexOf("."));
        if ("ssmone".equals(whichDataSource)) {
            DataSourceHolder.setDataSources(Constants.DATASOURCE_ONE);
        } else {
            DataSourceHolder.setDataSources(Constants.DATASOURCE_TWO);
        }
    }

    /**
     * 执行后将数据源置空
     */
    public void after() {
        DataSourceHolder.setDataSources(null);
    }

}
